import google.generativeai as genai
import os

# 确保你的 API 密钥已经设置为环境变量 GOOGLE_API_KEY
# 或者直接在这里配置
# genai.configure(api_key="YOUR_GOOGLE_API_KEY") 
genai.configure(api_key="AIzaSyBHLohL3JURhWWoWa28JVqHeFXDK7hNgrQ")


print("--- 可用的 Gemini 模型列表 (支持 'generateContent' 方法) ---")

try:
    # 遍历所有可用的模型
    for m in genai.list_models():
        # 检查模型是否支持我们需要的 'generateContent' 方法
        if 'generateContent' in m.supported_generation_methods:
            # 如果支持，就打印出它的名称
            print(m.name)
except Exception as e:
    print(f"查询模型列表时出错: {e}")

print("---------------------------------------------------------")
print("\n请从以上列表中选择一个合适的模型名称更新到您的 app.py 文件中。")